how to make python scripts more flexible